import time as _time
import torch as _torch
from src.networks import SimpleConvAttention as _net

_name = 'SimpleConvAtt'

# MODEL PARA
USE_NET = _net
###############


# DO NOT MODIFY
_time = _time.strftime('%m-%d %H:%M:%S', _time.localtime())
TRAIN_DIR = 'runs/%s/' % _name + _time
VAL_DIR = 'runs/%s/' % _name + _time
MODEL_NAME = '%s.pkl' % _name
device = _torch.device('cuda:0' if _torch.cuda.is_available() else 'cpu')
############

